package com.jojo.elementary.day7_linkedList;

import com.jojo.elementary.entity.ListNode;

/**
 * 203、移除链表元素
 *
 * 给你一个链表的头节点 head 和一个整数 val ，请你删除链表中所有满足 Node.val == val 的节点，并返回 新的头节点
 *
 * 示例 1：
 * 输入：head = [1,2,6,3,4,5,6], val = 6
 * 输出：[1,2,3,4,5]
 *
 * 示例 2：
 * 输入：head = [], val = 1
 * 输出：[]
 *
 * 示例 3：
 * 输入：head = [7,7,7,7], val = 7
 * 输出：[]
 *
 */
public class RemoveListNode {

    /** pe. 迭代 */
    public ListNode removeElements(ListNode head, int val) {
        ListNode newHead = new ListNode(0);
        newHead.next = head;
        ListNode temp = newHead;
        while (temp.next != null) {
            if (temp.next.val == val) {
                temp.next = temp.next.next;
            } else {
                temp = temp.next;
            }
        }
        return newHead.next;
    }

    /** myCode 递归 */
    public ListNode removeElements2(ListNode head, int val) {
        if (head == null){
            return null;
        }
        if (head.val == val){
            head = removeElements2(head.next,val);
        }else {
            head.next = removeElements2(head.next,val);
        }
        return head;
    }

    public static void main(String[] args) {
        ListNode list1 = new ListNode(1);
        ListNode list2 = new ListNode(2);
        ListNode list3 = new ListNode(6);
        ListNode list4 = new ListNode(3);
        ListNode list5 = new ListNode(4);
        ListNode list6 = new ListNode(5);
        ListNode list7 = new ListNode(6);
        list1.next = list2;
        list2.next = list3;
        list3.next = list4;
        list4.next = list5;
        list5.next = list6;
        list6.next = list7;
        RemoveListNode remove = new RemoveListNode();
        System.out.println(remove.removeElements2(list1, 6).val);
    }
}
